﻿using System;
using System.CodeDom;

using TechTalk.SpecFlow.Generator;
using TechTalk.SpecFlow.Generator.UnitTestProvider;

namespace CodeStream.Testing.WebAutomation.SpecFlow.Generator
{
    public class MsTest2010CodedUiGeneratorProvider : MsTest2010GeneratorProvider, IUnitTestGeneratorProvider
    {
        public override void SetTestClass(TestClassGenerationContext generationContext, string featureTitle, string featureDescription)
        {
            generationContext.TestClass.CustomAttributes.Add(
                new CodeAttributeDeclaration("Microsoft.VisualStudio.TestTools.UITesting.CodedUITestAttribute"));
        }
    }
}
